#pragma once
#include <vector>
#include <iostream>
#include <semaphore.h>

template <class T>
class RingQueue
{
private:
    void P(sem_t &sem)
    {
        sem_wait(&sem);
    }

    void V(sem_t &sem)
    {
        sem_post(&sem);
    }
public:
    RingQueue(int maxcap = 5)
        : _maxcap(maxcap), _ringqueue(maxcap), _p_step(0), _c_step(0)
    {
        sem_init(&_pspace_sem,0,maxcap);
        sem_init(&_cdata_sem,0,0);

        pthread_mutex_init(&_p_mutex,nullptr);
        pthread_mutex_init(&_c_mutex,nullptr);
    }

    void push(const T& in)
    {
        P(_pspace_sem);//信号量的申请本来就是原子的，所以加锁的时候就需要在这之后

        pthread_mutex_lock(&_p_mutex);
        _ringqueue[_p_step] = in;
        _p_step++;
        _p_step %= _maxcap;
        pthread_mutex_unlock(&_p_mutex);

        V(_cdata_sem);
    }

    void pop(T *out)
    {
        P(_cdata_sem);

        pthread_mutex_lock(&_c_mutex);
        *out = _ringqueue[_c_step];
        _c_step++;
        _c_step %= _maxcap;
        pthread_mutex_unlock(&_c_mutex);

        V(_pspace_sem);
    }

    ~RingQueue()
    {
        sem_destroy(&_pspace_sem);
        sem_destroy(&_cdata_sem);

        pthread_mutex_destroy(&_p_mutex);
        pthread_mutex_destroy(&_c_mutex);

    }

private:
    std::vector<T> _ringqueue; // 用vector模拟环形队列
    int _maxcap;               // 环形队列的最大容量

    int _p_step; // 生产者下标
    int _c_step; // 消费者下标

    sem_t _pspace_sem; // 生产者关注的空间资源
    sem_t _cdata_sem;  // 消费者关注的数据资源

    pthread_mutex_t _p_mutex;
    pthread_mutex_t _c_mutex;
};